# -*- coding: utf-8 -*-
"""
@Time    : 2024/7/17 19:58 
@Author  : ZhangShenao 
@File    : 1.使用arxiv工具开发科研助理.py
@Desc    : arXiv是一个提供免费访问的预印本库,供研究者在正式出版前上传和分享其研究工作
"""
import os

import dotenv
from langchain.agents import initialize_agent, AgentType
from langchain_community.agent_toolkits.load_tools import load_tools
from langchain_openai import ChatOpenAI

# 加载环境变量
dotenv.load_dotenv()

# 创建LLM
llm = ChatOpenAI(model_name='gpt-3.5-turbo', openai_api_base=os.getenv('OPENAI_API_BASE'))

# 加载arxiv工具
tools = load_tools(['arxiv'])

# 初始化代理链
agent_chain = initialize_agent(
    tools=tools,
    llm=llm,
    agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
    verbose=True,
)

# 执行代理链
agent_chain.run('请帮我介绍一下2005.14165这篇论文有哪些创新点？')
